simple prime number program in python